<?php
// +----------------------------------------------------------------------
// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
// +----------------------------------------------------------------------
// | Copyright (c) 2006-2014 http://thinkphp.cn All rights reserved.
// +----------------------------------------------------------------------
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
// +----------------------------------------------------------------------
// | Author: liu21st <liu21st@gmail.com>
// +----------------------------------------------------------------------
namespace Think\Cache\Driver;
use Think\Cache;
defined('THINK_PATH') or exit();

/**
 * Redis缓存驱动 
 * 要求安装phpredis扩展：https://github.com/nicolasff/phpredis
 */
class Redis extends Cache {
	 /**
	 * 架构函数
     * @param array $options 缓存参数
     * @access public
     */
    public function __construct($options=array()) {
        if ( !extension_loaded('redis') ) {
            E(L('_NOT_SUPPORT_').':redis');
        }
        if(empty($options)) {
            $options = array (
                'host'          => C('REDIS_HOST') ? C('REDIS_HOST') : '127.0.0.1',
                'port'          => C('REDIS_PORT') ? C('REDIS_PORT') : 6379,
                'timeout'       => C('DATA_CACHE_TIMEOUT') ? C('DATA_CACHE_TIMEOUT') : false,
                'persistent'    => false,
            );
        }
        $this->options =  $options;
        $this->options['expire'] =  isset($options['expire'])?  $options['expire']  :   C('DATA_CACHE_TIME');
        $this->options['prefix'] =  isset($options['prefix'])?  $options['prefix']  :   C('DATA_CACHE_PREFIX');        
        $this->options['length'] =  isset($options['length'])?  $options['length']  :   0;

        //新的redis连接方式
        $this->handler  = new \Redis;
        if($options['host'] === '127.0.0.1'){
            $this->handler->connect('/dev/shm/redis.sock');
        } else {
            $options['timeout'] === false ?
                $this->handler->pconnect($options['host'], $options['port']) :
                $this->handler->pconnect($options['host'], $options['port'], $options['timeout']);

        }
    }

    /**
     * 读取缓存
     * @access public
     * @param string $name 缓存变量名
     * @return mixed
     */
    public function get($name) {
        N('cache_read',1);
        $value = $this->handler->get($this->options['prefix'].$name);
        $jsonData  = json_decode( $value, true );
        return (is_array($jsonData) && $jsonData !== NULL) ? $jsonData : $value;	//检测是否为JSON数据 true 返回JSON解析数组, false返回源数据
    }

    /**
     * 写入缓存
     * @access public
     * @param string $name 缓存变量名
     * @param mixed $value  存储数据
     * @param integer $expire  有效时间（秒）
     * @return boolean
     */
    public function set($name, $value, $expire = null) {
        N('cache_write',1);
        if(is_null($expire)) {
            $expire  =  $this->options['expire'];
        }
        $name   =   $this->options['prefix'].$name;
        //对数组/对象数据进行缓存处理，保证数据完整性
        $value  =  (is_object($value) || is_array($value)) ? json_encode($value) : $value;
        if(is_int($expire) && $expire > 0) {
            $result = $this->handler->setex($name, $expire, $value);
        }else{
            $result = $this->handler->set($name, $value);
        }
        if($result && $this->options['length']>0) {
            // 记录缓存队列
            $this->queue($name);
        }
        return $result;
    }

    /**
     * 删除缓存
     * @access public
     * @param string $name 缓存变量名
     * @return boolean
     */
    public function rm($name) {
        return $this->handler->delete($this->options['prefix'].$name);
    }

    /**
     * 清除缓存
     * @access public
     * @return boolean
     */
    public function clear() {
        return $this->handler->flushDB();
    }
	
	/******************************add by ftxjoy*******************************/
    /**
     * 阻塞式读取 左为尾/右为头 队列第一条数据 并备份记录到队列尾
     * @access public
     * @param string $name  队列变量名
     * @return string $value 第一条值 
     */
    public function brpoplpush($name,$timeout=30) {
        return $this->handler->brpoplpush($this->options['prefix'].$name,$this->options['prefix'].$name,$timeout);
    }

    /**
     * 写入 左为尾/右为头 队列尾部数据
     * @access public
     * @param string $name  队列变量名
     * @param string $value 值
     * @return boolean
     */
    public function lpush($name,$value) {
        return $this->handler->lpush($this->options['prefix'].$name,$value);
    }

    /**
     * 写入 左为尾/右为头 队列头部数据
     * @access public
     * @param string $name  队列变量名
     * @param string $value 值
     * @return boolean
     */
    public function rpush($name,$value) {
        return $this->handler->rpush($this->options['prefix'].$name,$value);
    }
	
    /**
     * 删除 左为尾/右为头 队列尾部指定数据
     * @access public
     * @param string $name  队列变量名
     * @param string $value 值
     * @param int $count    删除数量
     * @return boolean
     */
    public function lrem($name,$value,$count) {
        return $this->handler->lrem($this->options['prefix'].$name,$value,$count);
    }
	
    /**
     * 读取队列范围数据
     * @access public
     * @param string $name  队列变量名
     * @param string $start 开始位置
     * @param string $end   结束位置
     * @return array $value 返回值 
     */
    public function lrange($name, $start=0, $end=-1) {
        return $this->handler->lrange($this->options['prefix'].$name,$start,$end);
    }
	
    /**
     * 变量自减
     * @access public
     * @param string $name  队列变量名
     * @return int          自减后值
     */
    public function decr($name) {
        return $this->handler->decr($this->options['prefix'].$name);
    }

    public function sadd($name,$value) {
        return $this->handler->sadd($this->options['prefix'].$name,$value);
    }

    public function srem($name,$value) {
        return $this->handler->srem($this->options['prefix'].$name,$value);
    }

    public function smembers($name) {
        return $this->handler->smembers($this->options['prefix'].$name);
    }

    public function zincrby($name,$i,$value) {
        return $this->handler->zincrby($this->options['prefix'].$name,$i,$value);
    }

    public function zrangebyscore($name,$min,$max) {
        return $this->handler->zrangebyscore($this->options['prefix'].$name,$min,$max);
    }

    public function zrange($name,$start,$stop) {
        return $this->handler->zrange($this->options['prefix'].$name,$start,$stop);
    }

    public function __call($method,$args){
        //调用缓存类型自己的方法
        if(method_exists($this->handler, $method)){
            isset($args[0]) AND $args[0] = $this->options['prefix'].$args[0];//key前缀拼接
            return call_user_func_array(array($this->handler,$method), $args);
        }else{
            E(__CLASS__.':'.$method.L('_METHOD_NOT_EXIST_'));
            return;
        }
    }
}
